package org.example;

import javax.swing.*;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

/**
 * @ClassName TestMenu
 * @Description
 * @Author zhangsw
 * @Date 2020/10/17 10:59
 * @Version 1.0
 */
public class TestMenu {

    public static void main(String[] args) {
        List<SysMenu> menus = new ArrayList<>();
        SysMenu menu = new SysMenu();
        menu.setId(1);
        menu.setName("系统管理");
        menu.setPid(0);
        menu.setType(0);
        menus.add(menu);
        SysMenu menu2 = new SysMenu();
        menu2.setId(2);
        menu2.setName("用户管理");
        menu2.setPid(1);
        menu2.setType(1);
        menus.add(menu2);
        SysMenu menu3 = new SysMenu();
        menu3.setId(3);
        menu3.setName("新增");
        menu3.setPid(2);
        menu3.setType(2);
        menus.add(menu3);
        System.err.println(menus);
        processMenus(menus);
//        System.err.println(menu);
    }

    private static void processMenus(List<SysMenu> menus) {
        List<SysMenu> menuList = menus;
        Iterator<SysMenu> it = menus.iterator();
        while (it.hasNext()) {
            SysMenu menu = it.next();
            for (SysMenu sysMenu : menuList) {
                if (menu.getId() == sysMenu.getPid()) {
                    if (menu.getMenus() == null) {
                        List<SysMenu> sysMenus = new ArrayList<>();
                        sysMenus.add(sysMenu);
                        menu.setMenus(sysMenus);
                    } else {
                        menu.getMenus().add(sysMenu);
                    }

                }
            }
        }
    }


}
